Edited code, added CSS