Cleaned up code, removed images not used, added more styling