Liitle Update for cleaner code flow