Rewrote a code to make it look nicer