Fixed a little bug and added more visual stuff