Fixes the looks of the code