Fixed image issue and added button stuff.