Fixed bug and added background