Login controller fix
Simple design fixes