MHR-23: Improve scheduler code.

Signed-off-by: Amlal El Mahrouss <amlal@el-mahrouss-logic.com>

1 files changed, 7 insertions, 5 deletions

diff --git a/Kernel/KernelKit/ProcessScheduler.hpp b/Kernel/KernelKit/ProcessScheduler.hpp
index 243857ae..9768d56a 100644
--- a/Kernel/KernelKit/ProcessScheduler.hpp
+++ b/Kernel/KernelKit/ProcessScheduler.hpp
@@ -252,23 +252,25 @@ namespace NewOS
 		{
 			return mTeam.AsArray().Count() > 0;
 		}
+
 		bool operator!()
 		{
 			return mTeam.AsArray().Count() == 0;
 		}
 
-		ProcessTeam& CurrentTeam()
-		{
-			return mTeam;
-		}
+	public:
+		ProcessTeam& CurrentTeam();
 
+	public:
 		SizeT Add(Ref<ProcessHeader>& headerRef);
 		bool  Remove(SizeT headerIndex);
 
+	public:
 		Ref<ProcessHeader>& GetCurrent();
 		SizeT				Run() noexcept;
 
-		static Ref<ProcessScheduler> Shared();
+	public:
+		static Ref<ProcessScheduler&> Shared();
 
 	private:
 		ProcessTeam mTeam;